package init

import (
	"fmt"
	"github.com/fsnotify/fsnotify"
	"github.com/spf13/viper"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"log"
	"member_srv/basic/appConfig"
	"time"
)

func init() {
	ConfigInit()
	MysqlInit()
}

func ConfigInit() {
	viper.SetConfigFile("D:\\gowork\\src\\zk\\zg4_zk3_cs\\member_srv\\config.yaml")
	viper.ReadInConfig()
	viper.Unmarshal(&appConfig.ConfigData)
	fmt.Println(appConfig.ConfigData)

	viper.WatchConfig()
	viper.OnConfigChange(func(e fsnotify.Event) {
		err := viper.Unmarshal(&appConfig.ConfigData)
		if err != nil {
			return
		}
		fmt.Println("配置文件更新成功", appConfig.ConfigData)
	})
}

func MysqlInit() {
	data := appConfig.ConfigData.Mysql
	var err error

	appConfig.Once.Do(func() {
		dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
			data.User, data.Passwd, data.Host, data.Port, data.Database)

		appConfig.DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
		if err != nil {
			panic(fmt.Sprintf("无法连接数据库: %v", err))
		} else {
			log.Println("Mysql数据库连接成功")
		}

		sqlDB, _ := appConfig.DB.DB()
		// SetMaxIdleConns 设置空闲连接池中连接的最大数量。
		sqlDB.SetMaxIdleConns(10)
		// SetMaxOpenConns 设置打开数据库连接的最大数量。
		sqlDB.SetMaxOpenConns(100)
		// SetConnMaxLifetime 设置了可以重新使用连接的最大时间。
		sqlDB.SetConnMaxLifetime(time.Hour)

		err := appConfig.DB.AutoMigrate()
		if err != nil {
			panic("数据库迁移失败")
		} else {
			log.Println("数据库迁移成功")
		}
	})
}
